define a function in python